import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.TimeUnit;

import org.postgresql.Driver;

/**
 * @作者 ： pwx5328113 @时间： 2023/4/11
 * @测试点: 打开快速负载均衡，关闭所有连接后，验证心跳线程是否全部正常退出
 **/
public class JDBC_QuikAutoBalance_Case0035 {
	public static void main(String[] args) throws Exception {
		// start前，args[0] 包含3个节点，其中一个节点异常
		System.out.println(new Date() + "----------start----------");
		System.out.println(Driver.getGSVersion());
		List<Connection> connections = new ArrayList<>();
		System.out.println(new Date() + "----------进行10个建连----------");
		String url1 = "jdbc:postgresql://" + args[0] + "/" + args[1] + "?" + args[4];
		for (int i = 0; i < 20; i++) {
			try {
				Connection connection = DriverManager.getConnection(url1, args[2], args[3]);
				connections.add(connection);
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
		System.out.println(new Date() + "----------sleep 85s----------");
		TimeUnit.SECONDS.sleep(85);
		// sleep过程中，10s后恢复异常节点（心跳线程20S进行一次检测），观测后台kill连接情况
		System.out.println(new Date() + "----------再次进行15个建连----------");
		for (int i = 0; i < 15; i++) {
			try {
				Connection connection = DriverManager.getConnection(url1, args[2], args[3]);
				connections.add(connection);
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
		System.out.println(new Date() + "----------sleep 10s----------");
		TimeUnit.SECONDS.sleep(10);
		// sleep过程中，数据库实际节点查询上个操作每个节点实际连接数
		System.out.println(new Date() + "----------connections close start----------");
		for (Connection connection : connections) {
			try {
				connection.close();
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
		System.out.println(new Date() + "----------connections close end----------");
		System.out.println(new Date() + "----------end----------");
	}
}
